package cn.tedu.v4;

import java.io.*;
import java.net.*;
import java.util.*;


/**
 * 多人聊天室 Version 4 多人聊天室---消息转发【广播】功能
 * 建议: 单线聊天，多人聊天，多人聊天+QQ模式  【消息转发功能】
 * 拓展功能: 建议增加第三个版本 Version 3
 *
 * 消息转发步骤: 服务器给所有客户端写出内容，客户端之间相互收到内容
 *  1.集合存储所有写出的客户端--服务器端写出给所有的客户端
 *  2.广播消息给所有客户端---broadcastMessage  广播留言
 *      注意:写在ClientHandler上,必须为同步方法
 *  3.将客户端输出流添加到allOut集合中【必须同步块】
 *      注意: 写在ClientHandler的构造器--所有客户端都具备功能: 消息转发
 *  4.广播所有消息给客户端
 *      注意: 写在ClientHandler的run---客户端写出的消息，读入方法转发出去
 *  5.客户端退出，从 allOut 集合中移除
 *     注意: 写在ClientHandler的run--finally 【一定异移除链接】
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=6666;
    //1.私有属性: 集合存储一个客户端链接，发送文本，转发给其他客户
    private List<PrintWriter> allOut=new ArrayList<>();

    public Server(){
        try {
            serverSocket=new ServerSocket(PORT);
        } catch (BindException e){
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败...");
        }
    }

    public void start()  {
        try {
            while (true){
            System.out.println("-------等待客户端链接------");
            Socket socket = serverSocket.accept();

            new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开");
        } finally {
            if (serverSocket!=null&&!serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }
        }
    }

    public static void main(String[] args)  {
        new Server().start();
    }
    //2.广播消息给所有客户端-----功能 broadcastMessage 广播留言
    //2.1 broadcastMessage 提供转发给客户端消息【加同步锁】
    //    简单理解: 来一个客户端只允许当前客户端转发消息
    //2.2 String message 参数: 转发消息---代表: 消息
    private synchronized void broadcastMessage(String message){
        //2.3 增强for:  allOut 存储所有客户端
        //    理解: 【来很多客户端，很多客户端遍历出来，消息转发出去】
        for (PrintWriter pw:allOut){  //allOut保存所有链接的客户端
            pw.println("消息广播"+message); //理解: 把消息转发出去
        }
    }
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();

            try {
                 br=new BufferedReader(new InputStreamReader
                        (socket.getInputStream(),"UTF8"));
                 pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter
                        (socket.getOutputStream(),
                                "UTF8")), true);
                 //3.所有客户端都具备消息转发
                //理解: 客户端写出文字，服务器端负责转发文字
                //监视同步对象 allOut---存所有客户端
                //客户端存在线程中---执行存在执行顺序不一致情况
                synchronized (allOut) {
                    allOut.add(pw);  //客户端写出文字，存集合中
                }

            } catch (IOException e) {
                System.out.println("客户端断开通讯~");
            }
        }

        public void run() {
            try {

            String str=null;
             new Thread(()->{
                 try{
                 String message;
                 while ((message=br.readLine())!=null){
                     System.out.println("客户端说: "+message);
                     System.out.println("服务器端请输入: ");

                     //4.广播消息给所有客户端
                     // broadcastMessage转发: 写出文本内容
                     // message 客户端发给服务器端的内容，通过服务器读入: message
                     // 转发给所有的客户端
                     broadcastMessage(message);
                 }
                 }catch (IOException e){
                     System.out.println("客户端读入失败!");
                 }
             }).start();
               Scanner input=new Scanner(System.in);
                while(true){
                    System.out.println(ip+"服务器请输入: ");
                    String say=input.nextLine();

                    pw.println("服务器说: "+say);
                }
            } catch (NullPointerException e) {
                System.out.println("IO流传输对象，有对象为null,请立马核对。"
                        +"建议: 查看try块中对象可能赋值为null");

            } finally {
                // 5.一旦有 客户端断开，客户端从 allOut 集合中移除  输出流
                   //  必须同步块，结束客户端不一致
                synchronized (allOut) {  //监视allOut集合当中有无客户端断开
                    allOut.remove(pw);  // allOut集合客户端断开链接，移除写出流
                }
                if (br!=null) {
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
                if (pw!=null) pw.close();
                if (socket!=null&&!socket.isClosed()){
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
            }
        }
    }
}
